# 1.准备环境

```
cd models/ops/
sh ./make.sh
python test.py
cd ../../
```


# 2.准备数据  

```
python preprocess_RGBTCC.py
python preprocess_DroneRGBT.py


rm -rf data/DroneRGBT/val/
cp data/DroneRGBT/test data/DroneRGBT/val -r

python predataset_RGBTCC.py
python predataset_DroneRGBT.py
```

# 3.训练  

```
wget https://github.com/whai362/PVT/releases/download/v2/pvt_v2_b3.pth


python train.py \
    --data-dir ./data/RGBTCC \
    --save-dir ./pth_RGBTCC \
    --lr 1e-5 \
    --val-start 20 \
    --device 0
	
	
python train.py \
    --data-dir ./data/DroneRGBT \
    --save-dir ./pth_DroneRGBT \
    --lr 1e-5 \
    --val-start 20 \
    --device 0

```

# 4.测试

```
python test.py \
    --data-dir ./data/RGBTCC \
    --save-dir ./pth_RGBTCC/0912-143855 \
    --model "best_model_74.10237789664417.pth" \
    --device 0
	
python test.py \
    --data-dir ./data/DroneRGBT \
    --save-dir ./pth_DroneRGBT/0912-143918 \
    --model "best_model_30.217222222222222.pth" \
    --device 0
```